package com.df.hitime.common.config;

import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

/**
 * @description: 类描述
 * @author: tigerlin
 * @createtime: 2018/11/30 17:52
 */
public class ImgUploadUtils {
    public static String uploadPic(MultipartFile imgFile, String localPath, String filename) {
        if (!imgFile.isEmpty()) {
            //生成uuid作为文件名称
            String uuid = UUID.randomUUID().toString().replaceAll("-", "");
            //获得文件类型（可以判断如果不是图片，禁止上传）
            String contentType = imgFile.getContentType();
            //获得文件后缀名
            String suffixName = contentType.substring(contentType.indexOf("/") + 1);
            //得到 文件名
            filename = uuid + "." + suffixName;
            //文件保存路径
            localPath += filename;
            try {
                imgFile.transferTo(new File(localPath));
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return filename;
    }
}
